package com.gujiayue.common.exception;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.dao.DataIntegrityViolationException;
import org.springframework.dao.DuplicateKeyException;
import org.springframework.data.redis.connection.PoolException;
import org.springframework.web.HttpRequestMethodNotSupportedException;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;
import org.springframework.web.multipart.MaxUploadSizeExceededException;
import org.springframework.web.servlet.NoHandlerFoundException;

import com.gujiayue.common.restful.RestfulRtnInfo;

/**
 * 异常处理器
 */
@RestControllerAdvice
public class UnifyExceptionHandler {

	private static final Logger logger = LoggerFactory.getLogger(UnifyExceptionHandler.class);

	@ExceptionHandler(NoHandlerFoundException.class)
	public RestfulRtnInfo handlerNoFoundException(Exception e) {
		logger.error(e.getMessage(), e);
		return new RestfulRtnInfo(404, "路径不存在，请检查路径是否正确");
	}

	@ExceptionHandler(DuplicateKeyException.class)
	public RestfulRtnInfo handleDuplicateKeyException(DuplicateKeyException e) {
		logger.error(e.getMessage(), e);
		return new RestfulRtnInfo(400, "数据库错误");
	}

	@ExceptionHandler(Exception.class)
	public RestfulRtnInfo handleException(Exception e) {
		logger.error(e.getMessage(), e);
		return new RestfulRtnInfo(400, "操作失败，" + e.getMessage());
	}

	@ExceptionHandler(HttpRequestMethodNotSupportedException.class)
	public RestfulRtnInfo HttpRequestMethodNotSupportedException(HttpRequestMethodNotSupportedException e) {
		StringBuffer sb = new StringBuffer();
		sb.append("不支持");
		sb.append(e.getMethod());
		sb.append("请求方法，");
		sb.append("支持以下");
		String[] methods = e.getSupportedMethods();
		if (methods != null) {
			for (String str : methods) {
				sb.append(str);
				sb.append("、");
			}
		}
		logger.error(sb.toString(), e);
		return new RestfulRtnInfo(405, "请求错误");
	}

	/**
	 * spring默认上传大小100MB 超出大小捕获异常MaxUploadSizeExceededException
	 */
	@ExceptionHandler(MaxUploadSizeExceededException.class)
	public RestfulRtnInfo handleMaxUploadSizeExceededException(MaxUploadSizeExceededException e) {
		logger.error(e.getMessage(), e);
		return new RestfulRtnInfo(400, "文件大小超出10MB限制, 请压缩或降低文件质量! ");
	}

	@ExceptionHandler(DataIntegrityViolationException.class)
	public RestfulRtnInfo handleDataIntegrityViolationException(DataIntegrityViolationException e) {
		logger.error(e.getMessage(), e);
		return new RestfulRtnInfo(400, "字段太长,超出数据库字段的长度");
	}

	@ExceptionHandler(PoolException.class)
	public RestfulRtnInfo handlePoolException(PoolException e) {
		logger.error(e.getMessage(), e);
		return new RestfulRtnInfo(400, "Redis 连接异常!");
	}

}
